package com.zxy.ziems.server.statistics.controller;

import com.zxy.ziems.server.constant.DateType;
import com.zxy.ziems.server.dto.DataQueryDto;
import com.zxy.ziems.server.statistics.pojo.vo.EnergyFlowTreeVO;
import com.zxy.ziems.server.statistics.service.EnergyFlowService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.xml.datatype.DatatypeConstants;

/**
 * 能流图
 */
@RestController
@RequestMapping("/api/energy/flow")
public class EnergyFlowController {

    @Autowired
    private EnergyFlowService energyFlowService;

    /**
     * 能流图
     * @param dataQueryDto
     * @return
     */
    @RequestMapping("/tree")
    public EnergyFlowTreeVO tree(DataQueryDto dataQueryDto) {

        if (dataQueryDto.getQueryType().equals(DateType.DAY)) {
            dataQueryDto.setQueryType(DateType.CUR_DAY);
        } else if (dataQueryDto.getQueryType().equals(DateType.MONTH)) {
            dataQueryDto.setQueryType(DateType.CUR_MONTH);
        }

        return energyFlowService.tree(dataQueryDto);
    }
}
